# --------------------------------------------------------------------
# Tests from the python/ray.workflow/tests directory.
# Covers all tests starting with `test_`.
# Please keep these sorted alphabetically.
# --------------------------------------------------------------------
load("//bazel:python.bzl", "py_test_module_list")

SRCS = [] + select({
    "@bazel_tools//src/conditions:windows": glob([
        # TODO(mehrdadn): This should be added for all platforms once resulting errors are fixed
        "**/conftest.py",
    ]),
    "//conditions:default": [],
})

LARGE_TESTS = ["tests/test_recovery.py"]

py_test_module_list(
  files = glob(["tests/test_*.py", "examples/**/*.py"], exclude=LARGE_TESTS),
  size = "medium",
  extra_srcs = SRCS,
  tags = ["team:core", "exclusive"],
  deps = ["//:ray_lib"],
)

py_test_module_list(
  files = LARGE_TESTS,
  size = "large",
  extra_srcs = SRCS,
  tags = ["team:core", "exclusive"],
  deps = ["//:ray_lib"],
)
